home *** CD-ROM | disk | FTP | other *** search
/ Visual Basic Source Code / Visual Basic Source Code.iso / vbsource / vbdatabs / ix_debug.cpp < prev    next >
C/C++ Source or Header  |  1999-03-31  |  9KB  |  339 lines
  1. // ------------------------------- //
  2. // -------- Start of File -------- //
  3. // ------------------------------- //
  4. // ----------------------------------------------------------- // 
  5. // C++ Source Code File Name: ix_debug.cpp 
  6. // Compiler Used: MSVC40, DJGPP 2.7.2.1, GCC 2.7.2.1, HP CPP 10.24
  7. // Produced By: Doug Gaer   
  8. // File Creation Date: 02/12/1997 
  9. // Date Last Modified: 03/31/1999
  10. // Copyright (c) 1997 Douglas M. Gaer
  11. // ----------------------------------------------------------- // 
  12. // ------------- Program Description and Details ------------- // 
  13. // ----------------------------------------------------------- // 
  14. /*
  15. The VBD C++ classes are copyright (c) 1997, by Douglas M. Gaer.
  16. All those who put this code or its derivatives in a commercial
  17. product MUST mention this copyright in their documentation for
  18. users of the products in which this code or its derivative
  19. classes are used. Otherwise, you have the freedom to redistribute
  20. verbatim copies of this source code, adapt it to your specific
  21. needs, or improve the code and release your improvements to the
  22. public provided that the modified files carry prominent notices
  23. stating that you changed the files and the date of any change.
  24.  
  25. THIS SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND.
  26. THE ENTIRE RISK OF THE QUALITY AND PERFORMANCE OF THIS SOFTWARE
  27. IS WITH YOU. SHOULD ANY ELEMENT OF THIS SOFTWARE PROVE DEFECTIVE,
  28. YOU WILL ASSUME THE COST OF ALL NECESSARY SERVICING, REPAIR, OR
  29. CORRECTION.
  30.  
  31. This program is used to debug B-tree index files created with
  32. the VBD file manager.
  33. */
  34. // ----------------------------------------------------------- // 
  35. #include <iostream.h>
  36. #include "btree.h"
  37. #include "btreeprt.h"
  38. #include "btwalk.h"
  39. #include "btwalker.h"
  40. #include "vbdstats.h"
  41.  
  42. // Program name and version number
  43. const char *VersionNumber = "1031.101";
  44. const char *ProgramName = "ix_debug";
  45.  
  46. // Default cache size for the disk based btree
  47. int cache_size = 1024;
  48.  
  49. void SkipToEol(istream &s)
  50. // Used to clear istream
  51. {
  52.   char c;
  53.   s.clear();
  54.   while(s.get(c) && c != '\n') { ; }
  55. }
  56.  
  57. void PrintKey(EntryKey &e)
  58. {
  59.   cout << e.key << " ";
  60. }
  61.  
  62. void Sort(Btree &btx)
  63. {
  64.   BtreeWalker btw(&btx);
  65.   unsigned num_objects = btw.Sort(PrintKey);
  66.   cout << endl;
  67.   cout << "Sorted " << num_objects << " objects" << endl;
  68.   cout << endl;
  69. }
  70.  
  71. void Walk(Btree &btx, int print_nodes = 0)
  72. {
  73.   char key;
  74.   BtreeWalkOrder walk_order;
  75.   
  76.   if (!cin) { // Input is in fail state
  77.     SkipToEol(cin); // Go to end of line
  78.     if (!cin) {  // Can't fix
  79.       cout << "Input stream is broken" << endl;
  80.       return;
  81.     }
  82.   }
  83.   cout << endl;
  84.   cout << "(f) Flat, (i) In-Order, (p) Pre-Order, (t) Post-Order > ";
  85.   cin >> key;
  86.   switch(key) {
  87.     case 'f' : case 'F' : walk_order = btLVLORDER; break;
  88.     case 'i' : case 'I' : walk_order = btINORDER; break;
  89.     case 'p' : case 'P' : walk_order = btPREORDER; break;
  90.     case 't' : case 'T' : walk_order = btPOSTORDER; break;
  91.     default:
  92.       cout << "Unrecognized key" << endl;
  93.       cout << endl;
  94.       return;
  95.   }
  96.  
  97.   CachePointer nxt(*btx.GetCache());
  98.   BtreeWalkerb tw(&btx, walk_order);
  99.   nxt = btx.GetRoot();
  100.  
  101.   int num_nodes = 0; int num_entries = 0;
  102.   while((__LWORD__)nxt != 0) {
  103.     nxt = tw.Next();
  104.     if((__LWORD__)nxt) {
  105.       num_nodes++;
  106.       int n = nxt->cnt;
  107.       if(print_nodes) PrintNode(nxt);
  108.       for(int i = 0; i < n; i++) {
  109.     if(print_nodes == 0) cout << nxt->entry[i].key << " ";
  110.     num_entries++;
  111.       }
  112.     }
  113.   }
  114.   cout << endl;
  115.   cout << num_nodes << " nodes in this tree" << endl;
  116.   cout << num_entries << " entries in this tree" << endl;
  117.   cout << endl;
  118. }
  119.  
  120. void Version()
  121. {
  122.   cout << endl;
  123.   cout << ProgramName << " version number: " << VersionNumber << endl;
  124.   cout << endl;
  125. }
  126.  
  127. void pause()
  128. {
  129.   cout << endl;
  130.   cout << "Press enter to continue..." << endl;
  131.   cin.get();
  132. }
  133.  
  134. char *InputData()
  135. {
  136.   char buf[255];
  137.   cout << "Enter Key Name: ";
  138.   cin.getline(buf, sizeof(buf));
  139.   unsigned len = 0;
  140.   for(unsigned i = 0; buf[i] != '\0'; i++) len++;
  141.   char *s = new char[len];
  142.   s[len] = '\0';
  143.   memmove(s, buf, len);
  144.   cout << endl;
  145.   return s;
  146. }
  147.  
  148. int Quit()
  149. {
  150.   cout << "Exiting..." << endl;
  151.   return 0;
  152. }
  153.  
  154. void FindKey(Btree &t)
  155. // Searchs the btree for a specified entry key
  156. {
  157.   char *buf = InputData();
  158.   int rv;
  159.  
  160.   if(!buf) {
  161.     cout << "Invalid input." << endl;
  162.     cout << endl;
  163.     return;
  164.   }
  165.   
  166.   EntryKey e = buf;
  167.   rv = t.Search(e);
  168.   
  169.   if(!rv) {
  170.     cout << "Could not find: " << buf << endl;
  171.     cout << endl;
  172.     return;
  173.   }
  174.   
  175.   cout << "Found: " << buf << endl;
  176.   cout << "Entry Key Name:             " << e.key << endl;
  177.   cout << "Object's data file address: " << e.object_address << endl;
  178.   cout << "Object's class ID:          " << e.class_id << endl;
  179.   cout << endl;
  180. }
  181.  
  182. void Menu(void)
  183. {
  184.   cout << "(D)   Dump the B-tree nodes" << endl;
  185.   cout << "(F)   Find a key in the B-tree index file" << endl;
  186.   cout << "(H)   Help (prints this menu)" << endl;
  187.   cout << "(Q)   Quit" << endl;
  188.   cout << "(R)   Dump the B-tree nodes recursively" << endl;
  189.   cout << "(S)   Sort the entry keys" << endl;
  190.   cout << "(T)   Display B-tree stats" << endl;
  191.   cout << "(V)   Display VBD file stats" << endl;
  192.   cout << "(W)   Walk the tree" << endl;
  193.   cout << endl;
  194. }
  195.  
  196. int main(int argc, char **argv)
  197. {
  198.   // Display the program version information and exit the program
  199.   if(argc >= 2) {
  200.     if(strcmp(argv[1], "version") == 0) {
  201.       Version();
  202.       return 0;
  203.     }
  204.   }
  205.  
  206.   if(argc < 2) {
  207.     cerr << endl;
  208.     cerr << "B-tree index file debug utility version "
  209.      << VersionNumber << endl;
  210.     cerr << "Usage: " << ProgramName << " infile.btx" << endl;
  211.     cerr << "Usage: " << ProgramName << " infile.btx (command)" << endl;
  212.     cerr << endl;
  213.     return 1;
  214.   }
  215.  
  216.   VBDFilePtr f(new VBDFile);
  217.   const char *FName = argv[1];
  218.   if(!VBDFile::Exists(FName)) 
  219.     Error->SignalException(EHandler::NoFileExists);
  220.   else
  221.     f->Open(FName, VBDFile::READONLY);
  222.  
  223.   FileHeader fh;
  224.   VBHeader vb;
  225.   
  226.   // Analyze the VBD file header to determine if the file has
  227.   // a pre-allocated static area
  228.   __LWORD__ static_area;
  229.   f->Read(&fh, sizeof(FileHeader), 0);
  230.  
  231.   // Check the HeapStart value 
  232.   f->Read(&vb, sizeof(VBHeader), fh.HeapStart);
  233.   if (vb.CkWord != CheckWord) {
  234.     cout << "Bad HeapStart value." << endl;
  235.     cout << "This VBD file is damaged." << endl;
  236.     cout << "Exiting..." << endl;
  237.     cout << endl;
  238.     return 0;
  239.   }
  240.   else    
  241.     // If no errors, calculate the the size of the static area
  242.     static_area = fh.HeapStart - sizeof(FileHeader);
  243.  
  244.   if(static_area == 0) {
  245.     cout << "No static area was alloacted in this file." << endl;
  246.     cout << "Cannot connect B-tree header." << endl;
  247.     cout << "Exiting..." << endl;
  248.     cout << endl;
  249.     return 0;
  250.   }
  251.  
  252.   FAU StaticEOF = f->FileSize(f->VBDFileName());
  253.   FAU addr = f->VBSearch(0); // Search the entire file
  254.   
  255.   if(addr == 0) {
  256.     cout << endl;
  257.     cout << "No variable blocks found in file: "
  258.      << f->VBDFileName() << endl;
  259.     cout << "Exiting..." << endl;
  260.     cout << endl;
  261.     return 0;
  262.   }
  263.  
  264.   Btree t(cache_size);
  265.   t.Connect(f, 0);
  266.  
  267.   // Main menu
  268.   char key;
  269.   if(argc <= 2) Menu(); // Not processing a command
  270.   int rv = 1;
  271.   
  272.   while(rv) {
  273.     if(argc > 2) { // Process a single command and exit the loop
  274.       key = *(argv[2]);
  275.       rv = 0;
  276.     }
  277.     else {
  278.       if (!cin) { // Input is in fail state
  279.     SkipToEol(cin); // Go to end of line
  280.     if (!cin) {  // Can't fix
  281.           cout << "Input stream is broken" << endl;
  282.           return 0;
  283.     }
  284.       }
  285.       cout << '>';
  286.       cin >> key;
  287.       if (!cin) continue; // Fix at top of loop
  288.     }
  289.     switch(key) {
  290.       case 'd' : case 'D' :
  291.     if(argc <= 2) SkipToEol(cin);
  292.     Walk(t, 1);
  293.     break;
  294.       case 'f' : case 'F' :
  295.     if(argc <= 2) SkipToEol(cin);
  296.     FindKey(t);
  297.     break;    
  298.       case 'h' : case 'H' :
  299.     Menu();
  300.     break;
  301.       case '?' :
  302.     Menu();
  303.     break;
  304.       case 's' : case 'S' :
  305.     if(argc <= 2) SkipToEol(cin);
  306.     Sort(t);
  307.     break;
  308.       c